/*
   Copyright (c) 2023 李伟国
   learner-platform is licensed under Mulan PSL v2.
   You can use this software according to the terms and conditions of the Mulan PSL v2.
   You may obtain a copy of Mulan PSL v2 at:
            http://license.coscl.org.cn/MulanPSL2
   THIS SOFTWARE IS PROVIDED ON AN "AS IS" BASIS, WITHOUT WARRANTIES OF ANY KIND, EITHER EXPRESS OR IMPLIED, INCLUDING BUT NOT LIMITED TO NON-INFRINGEMENT, MERCHANTABILITY OR FIT FOR A PARTICULAR PURPOSE.
   See the Mulan PSL v2 for more details.
*/

package cn.platform.utils;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import java.io.File;
import java.io.FileWriter;
import java.io.IOException;

public class FileUtil {

    private static Logger logger = LoggerFactory.getLogger(FileUtil.class);

    /**
     * 删除文件或文件夹
     *
     * @param folderPath 文件或文件夹的路径
     */
    public static boolean delete(String folderPath) {
        boolean result = true;
        File folder = new File(folderPath);
        if (folder.exists()) { // 判断文件是否存在
            if (folder.isDirectory()) { // 判断是否是文件夹
                deleteFolder(folder.getPath()); // 递归删除文件夹下的所有文件和文件夹
            } else {
                result = folder.delete(); // 如果不是文件夹，直接删除该文件
            }
        }
        return result;
    }

    /**
     * 删除文件夹
     *
     * @param folderPath 文件夹路径
     */
    public static boolean deleteFolder(String folderPath) {
        File folder = new File(folderPath);
        // 首先判断要删除的文件夹是否存在或者是一个文件夹
        if (!folder.exists() || !folder.isDirectory()) {
            return false;
        }
        // 获取该文件夹下的所有文件和子目录，并使用 for 循环遍历每一个文件和子目录
        File[] files = folder.listFiles();
        boolean result = true;
        for (File file : files) {
            if (file.isDirectory()) {
                // 如果遍历到的是子目录，则递归调用本方法删除该子目录下的所有文件
                result &= deleteFolder(file.getAbsolutePath());
            } else {
                // 如果是文件，则直接删除该文件
                result &= file.delete();
            }
        }
        // 最后再删除该文件夹本身
        result &= folder.delete();
        // 最后返回一个布尔值，表示删除是否成功
        return result;
    }

    /**
     * 删除指定目录下的所有文件（保留该目录）
     *
     * @param folderPath 目录路径
     * @return 是否删除成功
     */
    public static boolean deleteFilesInFolder(String folderPath) {
        File folder = new File(folderPath);
        // 首先判断要删除的文件夹是否存在或者是一个文件夹
        if (!folder.exists() || !folder.isDirectory()) {
            return false;
        }
        // 获取该文件夹下的所有文件和子目录，并使用 for 循环遍历每一个文件和子目录
        File[] files = folder.listFiles();
        if(files == null){
            return true;
        }
        boolean result = true;
        for (File file : files) {
            if (file.isDirectory()) {
                // 如果遍历到的是子目录，则递归调用本方法删除该子目录下的所有文件
                result &= deleteFolder(file.getAbsolutePath());
            } else {
                // 如果是文件，则直接删除该文件
                result &= file.delete();
            }
        }
        // 最后返回一个布尔值，表示删除是否成功
        return result;
    }

    /**
     * 创建文件并向其写入数据
     *
     * @param fileName 文件名
     * @param data     要写入文件的文本内容
     * @return 是否成功写入文件
     */
    public static boolean writeFile(String fileName, String data) {
        try {
            File file = new File(fileName);  // 创建一个新文件
            if(!file.getParentFile().exists()){
                boolean mkdirs = file.getParentFile().mkdirs();
                if(!mkdirs){
                    logger.error("创建父级文件夹失败");
                }
            }
            if (!file.exists()) {  // 判断文件是否存在
                boolean newFile = file.createNewFile();// 创建新文件
                if(!newFile){
                    logger.error("创建新文件失败");
                }
            }
            FileWriter writer = new FileWriter(file);  // 创建一个FileWriter对象
            writer.write(data);  // 向文件中写入数据
            writer.close();  // 关闭文件流
            return true;
        } catch (IOException e) {
            logger.error(e.getLocalizedMessage());
            return false;
        }
    }

}
